package collection._05TreeSet;

import java.util.TreeSet;

/*
TreeSet底层是红黑树
 */
public class Demo01 {
    public static void main(String[] args) {
        /**
         * 需求：
         *      利用TreeSet存储整数并进行排序
         */
        //1.创建TreeSet集合对象
        TreeSet treeSet = new TreeSet<>();

        //2.添加元素
        /*treeSet.add(1);
        treeSet.add(2);
        treeSet.add(3);
        treeSet.add(4);
        treeSet.add(5);
        treeSet.add(6);
        treeSet.add(7);
        treeSet.add(8);
*/
        //3.打印集合
        //System.out.println(treeSet);

        //遍历集合 ：Iterator 增强for Lambda

        Student s1 = new Student("zhangsan",21);
        Student s2 = new Student("lisi",22);
        Student s3 = new Student("wangwu",23);

        /**
         * 方式1：默认的排序规则/自然排序
         * Student实现Comparable接口，重写里面的抽象方法
         *
         */
        treeSet.add(s1);
        treeSet.add(s2);
        treeSet.add(s3);
        System.out.println(treeSet);






    }
}
class Student implements Comparable<Student> {
    private String name;
    private int age;

    public Student() {
    }

    public Student(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }


    @Override
    /*
    this 表示当前要添加的元素
    o 表示红黑树中存在的元素

    返回值：负数 -> 当前要添加的元素是小的，存左边
          正数 -> 当前要添加的元素是大的，存右边
           0  -> 当前要添加的元素已经存在，舍弃
     */
    public int compareTo(Student o) {
        //指定排序的规则
        //按照年龄的升序进行排序

        return this.getAge() - o.getAge();
    }
}
